from time import sleep
from basepage import BasePage
from selenium.webdriver.common.by import By
from Python基础软件测试课程.APP自动化.screenshot import Screen
class RetrievePasswordPage(BasePage):
	# 初始化，用于实例化Screen类
    def __init__(self, driver):
        super().__init__(driver)
        self.screen = Screen(driver)
	# 自动获取图片验证码中的内容
    def auto_get_check_code(self, telephone):
        self.input_telephone(telephone)
        # 获取图片验证码，获取到的是元组
        check_code_tuple = self.screen.get_part_screen()
        print(check_code_tuple)
        # 在元组中提取第一个数据(元组下标位置从0开始)，获取到的数据为字典（字典是一组键值对）
        check_code_list = check_code_tuple[0]
        print(check_code_list)
        # 在字典中提取第三个数据'words_result'，获取到的数据为列表
        word_result = check_code_list['words_result']
        # 判断获取到的列表的值是否为空，为空则在列表中自定义一个字典“word_result”的值
        if word_result:
            print(word_result)
        else:
            word_result = [{'words': 'F3HA'}]
        print(word_result)
        # 在列表中提取一个数据，获取到的数据是字典
        dic_words = word_result[0]
        print(dic_words)
        # 在字典中输出"words"
        word = dic_words['words']
        print(word)
        self.input_image_check_code(word)
        self.find_xpath(self._btn_confirm).click()
        sleep(1)